package com.yingdong.redis;

import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;
import java.util.concurrent.locks.ReentrantLock;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

public class RedisManager {
	private static JedisPool pool = null;
	private static ReentrantLock lock = new ReentrantLock();

	public static void init() throws IOException {
		if (pool == null) {
			Properties prop = new Properties();
			InputStream in = Thread.currentThread().getContextClassLoader().getResourceAsStream("jedis.properties");
			prop.load(in);
			JedisPoolConfig config = new JedisPoolConfig();
			// 控制一个pool可分配多少个jedis实例，通过pool.getResource()来获取；
			// 如果赋值为-1，则表示不限制；如果pool已经分配了maxActive个jedis实例，则此时pool的状态为exhausted(耗尽)。
			config.setMaxTotal(Integer.decode(prop.getProperty("jedis.maxtotal")));
			// 控制一个pool最多有多少个状态为idle(空闲的)的jedis实例。
			config.setMaxIdle(Integer.decode(prop.getProperty("jedis.maxidle")));
			// 表示当borrow(引入)一个jedis实例时，最大的等待时间，如果超过等待时间，则直接抛出JedisConnectionException；
			config.setMaxWaitMillis(Integer.decode(prop.getProperty("jedis.maxwaitmillis")));
			// 在borrow一个jedis实例时，是否提前进行validate操作；如果为true，则得到的jedis实例均是可用的；
			config.setTestOnBorrow(true);
			pool = new JedisPool(config, prop.getProperty("jedis.ip"), Integer.decode(prop.getProperty("jedis.port")));
		}
	}

	/**
	 * 返还到连接池
	 * 
	 * @param redis
	 */
	public static void returnResource(Jedis redis) {
		try {
			lock.lock();
			if (redis != null) {
				pool.returnResourceObject(redis);
			}
		} finally {
			lock.unlock();
		}
	}

	/**
	 * 获得连接池
	 * 
	 * @param pool
	 * @param redis
	 */
	public static RedisClient getResource() {

		try {
			lock.lock();
			return new RedisClient(pool.getResource());
		} finally {
			lock.unlock();
		}

	}

}
